package com.qcby.teacher.config;

import java.io.IOException;
import com.lowagie.text.Document;
import com.lowagie.text.DocumentException;
import com.lowagie.text.Paragraph;
import com.lowagie.text.pdf.PdfWriter;
import com.qcby.teacher.pdf.MarkdownDTO;

import java.io.ByteArrayOutputStream;
import java.util.List;

public class TeachMessage {
    public static String getTeachMessage(String answer) throws IOException {
        // 假设您的数据格式是固定的，可以根据特定的格式来提取需要的信息
        String result = null;

        // 在这里编写代码来提取您需要的信息
        // 例如，从字符串中找到特定的标记来确定需要提取的信息

        // 示例：假设您的数据中包含 "result" 字段，并且以 "result":" 开头，以 " 结尾
        int startIndex = answer.indexOf("\"result\":\"");
        if (startIndex != -1) {
            startIndex += "\"result\":\"".length();
            int endIndex = answer.indexOf("\"", startIndex);
            if (endIndex != -1) {
                result = answer.substring(startIndex, endIndex);
            }
        }

        return result;
    }
    public static byte[] generatePdfFromMarkdowns(List<MarkdownDTO> dtos) throws DocumentException {
        Document document = new Document();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        PdfWriter.getInstance(document, baos);
        document.open();

        for (MarkdownDTO dto : dtos) {
            document.add(new Paragraph(dto.getContent()));
        }

        document.close();
        return baos.toByteArray();
    }
}
